from datetime import datetime, timezone

import bitmapist


def test_from_date_year():
    ev1 = bitmapist.YearEvents.from_date(
        "foo", datetime(2014, 1, 1, tzinfo=timezone.utc)
    )
    ev2 = bitmapist.YearEvents("foo", 2014)
    assert ev1 == ev2


def test_from_date_month():
    ev1 = bitmapist.MonthEvents.from_date(
        "foo", datetime(2014, 1, 1, tzinfo=timezone.utc)
    )
    ev2 = bitmapist.MonthEvents("foo", 2014, 1)
    assert ev1 == ev2


def test_from_date_week():
    ev1 = bitmapist.MonthEvents.from_date(
        "foo", datetime(2014, 1, 1, tzinfo=timezone.utc)
    )
    ev2 = bitmapist.MonthEvents("foo", 2014, 1)
    assert ev1 == ev2


def test_from_date_day():
    ev1 = bitmapist.DayEvents.from_date(
        "foo", datetime(2014, 1, 1, tzinfo=timezone.utc)
    )
    ev2 = bitmapist.DayEvents("foo", 2014, 1, 1)
    assert ev1 == ev2


def test_from_date_hour():
    ev1 = bitmapist.HourEvents.from_date(
        "foo", datetime(2014, 1, 1, 1, tzinfo=timezone.utc)
    )
    ev2 = bitmapist.HourEvents("foo", 2014, 1, 1, 1)
    assert ev1 == ev2
